package com.haifeng.jupiter.common.core.base;

import cn.hutool.http.HttpStatus;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.io.Serializable;

/**
 * <p>
 * 响应信息主体
 * </p>
 *
 * @author: Haifeng
 * @date: 2020-08-09
 */
@Data
@ApiModel("响应信息主体")
@AllArgsConstructor
@NoArgsConstructor
public class Rest<T> implements Serializable {

    private static final long serialVersionUID = 1L;

    /**
     * 请求是否成功：true-成功；false-失败
     */
    @ApiModelProperty("请求是否成功：true-成功；false-失败")
    private Boolean success = false;

    /**
     * 数据
     */
    @ApiModelProperty("数据")
    private T data;

    /**
     * 状态码
     */
    @ApiModelProperty("状态码")
    private Integer code;

    /**
     * 返回信息
     */
    @ApiModelProperty("返回信息")
    private String message;

    /**
     * 默认成功返回
     * @return
     */
    public static Rest success() {
        return restResult(true, null, HttpStatus.HTTP_OK, "success");
    }

    /**
     * 指定data成功返回
     * @param data 数据
     * @return
     */
    public static Rest success(Object data) {
        return restResult(true, data, HttpStatus.HTTP_OK, "success");
    }

    /**
     * 指定data、message成功返回
     * @param data 数据
     * @param message 消息
     * @return
     */
    public static Rest success(Object data, String message) {
        return restResult(true, data, HttpStatus.HTTP_OK, message);
    }

    /**
     * 默认失败返回
     * @return
     */
    public static Rest error() {
        return restResult(false, null, HttpStatus.HTTP_INTERNAL_ERROR, "error");
    }

    /**
     * 带code、message失败返回
     * @param code 状态码
     * @param message 消息
     * @return
     */
    public static Rest error(int code, String message) {
        return restResult(false, null, code, message);
    }

    /**
     * 带data失败返回
     * @param data 数据
     * @return
     */
    public static Rest error(Object data) {
        return restResult(false, data, HttpStatus.HTTP_INTERNAL_ERROR, "error");
    }

    /**
     * 带data、code、message失败返回
     * @param data 数据
     * @param code 状态码
     * @param message 消息
     * @return
     */
    public static Rest error(Object data, int code, String message) {
        return restResult(false, data, code, message);
    }

    /**
     * 响应处理
     * @param success 是否成功
     * @param data 数据
     * @param code 状态码
     * @param message 消息
     * @return
     */
    private static Rest restResult(Boolean success, Object data, int code, String message) {
        Rest rest = new Rest<>();
        rest.setSuccess(success);
        rest.setCode(code);
        rest.setData(data);
        rest.setMessage(message);
        return rest;
    }
}
